package com.xuecheng.content.service.impl;

import com.xuecheng.content.mapper.CourseCategoryMapper;
import com.xuecheng.content.model.dto.CourseCategoryTreeDto;
import com.xuecheng.content.service.CourseCategoryService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
@Service
public class CourseCategoryServiceImpl implements CourseCategoryService {
    @Autowired
    CourseCategoryMapper courseCategoryMapper;

    //TODO 递归查询出分类信息(树状查询)
    @Override
    public List<CourseCategoryTreeDto> queryTreeNodes(String id) { //这里默认 传入的是1 也就是根节点
        // 调用mapper 递归查询出分类信息
        List<CourseCategoryTreeDto> courseCategoryTreeDtos = courseCategoryMapper.selectTreeNodes(id);

        // 找到每个节点的子节点 List<CourseCategoryTreeDto>
        // 现将list 转成 map key就是节点id，value就是CourseCategoryTreeDto对象，目的就是为了方便从map获取节点
        Map<String, CourseCategoryTreeDto> mapTemp = courseCategoryTreeDtos.stream()
                // 排除根节点传入的参数(id)是1
                .filter(item -> !id.equals(item.getId()))
                .collect(Collectors.toMap(CourseCategoryTreeDto::getId, value -> value, (key1, key2) -> key2));

        // 定义一个list 作为最终返回的list 这里面 存放的是 1-1  1-10  1-11  1-12  等等 一级目录
        List<CourseCategoryTreeDto> courseCategoryList = new ArrayList<>();

        // 从头遍历 List<CourseCategoryTreeDto> 一边遍历一遍找子节点放在父节点的 childrenTreeNodes
        courseCategoryTreeDtos.stream()
                .filter(item -> !id.equals(item.getId()))
                .forEach(item -> {
                    // 向list写入元素
                    if (item.getParentid().equals(id)) { // 如果 parentID = 1的话  说明他就是一级目录
                        courseCategoryList.add(item);
                    }

                    // 找到当前节点的父节点       当前节点:1-1-1  1-1-10  1-1-2  1-1-3 等等 二级节点   父节点则是 1-1  1-10  1-11  1-12  等等 一级目录
                    CourseCategoryTreeDto courseCategoryParent = mapTemp.get(item.getParentid());
                    if (courseCategoryParent != null) {
                        if (courseCategoryParent.getChildrenTreeNodes() == null) {
                            // 如果该父节点的属性 ChildrenTreeNodes 为空 我们需要new一个 容器(集合),因为我们要向该集合中存放它的子节点
                            courseCategoryParent.setChildrenTreeNodes(new ArrayList<CourseCategoryTreeDto>());
                        }
                        // 找到每个节点的子节点 放在父节点的 ChildrenTreeNodes 中
                        // 将 1-1-1  1-1-10  1-1-2  1-1-3 等数据 存放到 1-1节点对象 的 ChildrenTreeNodes中
                        courseCategoryParent.getChildrenTreeNodes().add(item);
                    }

                });

        return courseCategoryList;
    }

}
